


option(HOME "use libtorch code" ON)
if(HOME)
    include_directories ("${CMAKE_CURRENT_SOURCE_DIR}/include")
    include_directories ("D:/opencv455_cuda/install/include")
    include_directories ("D:/opencv455_cuda/install/include/opencv2")
    include_directories ("D:/libtorch260_debug/include")
    include_directories ("D:/libtorch260_debug/include/torch/csrc/api/include")

    link_libraries("D:/opencv455_cuda/install/x64/vc16/lib/opencv_core455.lib")
    link_libraries("D:/opencv455_cuda/install/x64/vc16/lib/opencv_imgcodecs455.lib")
    link_libraries("D:/opencv455_cuda/install/x64/vc16/lib/opencv_imgproc455.lib")
    link_libraries("D:/opencv455_cuda/install/x64/vc16/lib/opencv_highgui455.lib")


    set(SOURCE_FILES
            code_dataset_new.h
            captcha_recognize_new.h
            model_new.h
            captcha_recognize_new.cpp
            main_new.cpp
    )
    add_executable(lstm_captcha ${SOURCE_FILES})
else()
    include_directories ("${CMAKE_CURRENT_SOURCE_DIR}/include")
    include_directories ("E:/opencv452_cuda/install/include")
    include_directories ("E:/opencv452_cuda/install/include/opencv2")
    include_directories ("E:/libtorch251_debug/include")
    include_directories ("E:/libtorch251_debug/include/torch/csrc/api/include")

    link_libraries("E:/opencv452_cuda/install/x64/vc16/lib/opencv_core452.lib")
    link_libraries("E:/opencv452_cuda/install/x64/vc16/lib/opencv_imgcodecs452.lib")
    link_libraries("E:/opencv452_cuda/install/x64/vc16/lib/opencv_imgproc452.lib")
    link_libraries("E:/opencv452_cuda/install/x64/vc16/lib/opencv_highgui452.lib")


    set(SOURCE_FILES
            main.cpp
            include/code_dataset.h
            include/model.h
            include/captcha_recognize.h
            src/captcha_recognize.cpp
    )
    add_executable(lstm_captcha ${SOURCE_FILES})
endif()

